#!/usr/bin/env python
# -*- coding: UTF-8 -*-
"""
@Project : POS_api_test_project 
@File    : get_all_files_path.py
@IDE     : PyCharm 
@Author  : Mario
@Date    : 2022/5/26 17:53 
@Describe: 获取所有Data目录下的所有yaml文件
"""
import os


def get_all_files(file_path, yaml_data_switch=False) -> list:
    """
    获取文件路径
    :param file_path: 目录路径
    :param yaml_data_switch: 是否过滤文件为 yaml格式， True则过滤
    :return:
    """
    file_name = []
    # 获取所有文件下的子文件名称
    for root, dirs, files in os.walk(file_path):
        for filePath in files:
            path = os.path.join(root, filePath)
            if yaml_data_switch:
                if 'yaml' in path or '.yml' in path:
                    file_name.append(path)
            else:
                file_name.append(path)
    return file_name
